package org.zhengbo.imgUpload.model;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.util.Random;

import javax.imageio.ImageIO;
import javax.imageio.stream.ImageOutputStream;

/**
 * 驗證碼生成工具類
 * @author f8110026
 *
 */
public class CheckCode {
	private ByteArrayInputStream image;
	private String code;
	private BufferedImage bufferedImage;

	private static final char[] randomSequence = new char[] { 'A', 'B', 'C',
			'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P',
			'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '0', '1', '2',
			'3', '4', '5', '6', '7', '8', '9' };

	public CheckCode() {
		init();
	}

	public static CheckCode Instance() {
		return new CheckCode();
	}

	public ByteArrayInputStream getImage() {
		return this.image;
	}

	public String getCode() {
		return code;
	}

	public BufferedImage getBufferedImage() {
		return bufferedImage;
	}

	private void init() {

		int width = 80, height = 30;
		BufferedImage image = new BufferedImage(width, height,
				BufferedImage.TYPE_INT_RGB);
		bufferedImage = image;
		Graphics g = image.getGraphics();
		Random random = new Random();
		g.setColor(getRandColor(200, 250));
		g.fillRect(0, 0, width, height);
		Font font = new Font("Times New Roman", Font.PLAIN, height - 2);
		g.setFont(font);
		for (int i = 0; i < 155; i++) {
			g.setColor(getRandColor(160,200));
			int x = random.nextInt(width);
			int y = random.nextInt(height);
			int xl = random.nextInt(12);
			int yl = random.nextInt(12);
			g.drawLine(x, y, x + xl, y + yl);
		}
		String sRand = "";
		for (int i = 0; i < 4; i++) {
			font = new Font("Times New Roman", Font.PLAIN, height  + random.nextInt(10)- 10);
			g.setFont(font);
			int index = random.nextInt(35);
			String rand = String.valueOf(randomSequence[index]);
			sRand += rand;
			g.setColor(new Color(20 + random.nextInt(110), 20 + random
					.nextInt(110), 20 + random.nextInt(110)));
			g.drawString(rand, 15 * i + 6, 25 + random.nextInt(7) - 5);
		}
		this.code = sRand;
		g.dispose();
		ByteArrayInputStream input = null;
		ByteArrayOutputStream output = new ByteArrayOutputStream();
		try {
			ImageOutputStream imageOut = ImageIO
					.createImageOutputStream(output);
			ImageIO.write(image, "JPEG", imageOut);
			imageOut.close();
			input = new ByteArrayInputStream(output.toByteArray());
		} catch (Exception e) {
			System.out.println("驗證碼產生錯誤：" + e.toString());
		}

		this.image = input;
	}

	private Color getRandColor(int fc, int bc) {
		Random random = new Random();
		if (fc > 255)
			fc = 255;
		if (bc > 255)
			bc = 255;
		int r = fc + random.nextInt(bc - fc);
		int g = fc + random.nextInt(bc - fc);
		int b = fc + random.nextInt(bc - fc);
		return new Color(r, g, b);
	}

}
